FROM registry.cn-shanghai.aliyuncs.com/tcc-public/pytorch:1.11-cuda11.3-py3
# 安装python依赖包
RUN apt-get update && apt-get install g++ -y

# torch-geometric
RUN pip install scipy==1.10.1 scikit-learn==1.2.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN pip install torch-scatter==2.1.1 torch-sparse==0.6.13 torch-cluster==1.6.1 torch-spline-conv==1.2.2 torch-geometric==2.2.0 -f https://data.pyg.org/whl/torch-1.11.0+cu113.html

# 其他依赖
RUN pip install tqdm==4.62.3 pandas==2.0.0 numpy==1.21.2 -i https://pypi.tuna.tsinghua.edu.cn/simple

# 增加代码
ADD src.tar.gz /workspace

CMD ["bash", "run.sh"]

